Follow the Gradient: Tracing Safety Degradation Back to Training Data
跟随梯度:将安全退化追溯到训练数据
🧭 LLM 微调数据安全筛选
DatologyAI · University of Illinois Urbana-Champaign (UIUC)
SimpleLoRIF 将低秩梯度归因扩至百万级数据,过滤梯度对齐不安全的样本,攻击成功率最高降22%
📌 工作参考价值(4 点)
- 在微调推荐/Agent LLM 时,可用“梯度方向对齐目标/非目标行为”的分数筛选训练样本,去除与广告合规违规、品牌安全、诱导点击等不良输出梯度对齐的隐式风险数据,不依赖文本表面。
- 工程实现上,只对最后 4 个 transformer blocks 回传并配合低秩投影计算梯度 embedding,可处理百万级样本;推荐团队可用类似方法对商品/query 日志做安全与质量滤除,无需全模型回传。
- 过滤比例与能力存在平衡:删除 top 20% 最对齐样本可显著降低攻击成功率且能力保持或提升;业务上可扫描过滤比例阈值,评估线上指标。
- 可构建定制化数据过滤 pipeline,按比例或子集过滤风险数据;类似思路可迁移到搜索词改写、推荐理由生成等场景的合规性治理。